""" 从头文件中获取类名，私有成员变量，共有成员函数 """
import os
import re
if __name__ == "__main__":
    # 获取当前目录
    current_dir = os.getcwd()
    print(current_dir)
    # 获取当前目录下的所有文件
    files = os.listdir(current_dir)
    # 遍历所有文件
    for file in files:
        # 获取文件名
        file_name = os.path.splitext(file)[0]
        # 获取文件后缀
        file_ext = os.path.splitext(file)[1]
        # 判断是否是头文件
        if file_ext == ".h":
            # 读取头文件内容
            with open(file, "r") as f:
                content = f.read()
                # 获取类名
                class_name = re.findall(r"class\s+(\w+)\s*:", content)
                # 获取私有成员变量
                private_members = re.findall(r"private:\s*(\w+\s+\w+;)", content)
                # 获取共有成员函数
                public_methods = re.findall(r"public:\s*(\w+\s+\w+\(.*\);)", content)
                # 打印类名
                print("class name: ", class_name)
                # 打印私有成员变量
                print("private members: ", private_members)
                # 打印共有成员函数
                print("public methods: ", public_methods)
                # 打印空行
                print()